import PizZip from "pizzip";
import Docxtemplater from "docxtemplater";
import JSZipUtils from "jszip-utils";
import { saveAs } from "file-saver";
import docxtemplaterImage from "docxtemplater-image-module-free";

// 将图片地址转为base64，导出word图片只能是base64
export function base64DataURLToArrayBuffer(dataURL) {
  const base64Regex = /^data:image\/(png|jpg|svg|svg\+xml);base64,/;
  if (!base64Regex.test(dataURL)) {
    return false;
  }
  const stringBase64 = dataURL.replace(base64Regex, "");
  let binaryString;
  if (typeof window !== "undefined") {
    binaryString = window.atob(stringBase64);
  } else {
    binaryString = new Buffer(stringBase64, "base64").toString("binary");
  }
  const len = binaryString.length;
  const bytes = new Uint8Array(len);
  for (let i = 0; i < len; i++) {
    const ascii = binaryString.charCodeAt(i);
    bytes[i] = ascii;
  }
  return bytes.buffer;
}

export function getBase64Sync(imgUrl) {
  return new Promise(function (resolve, reject) {
    // 一定要设置为let，不然图片不显示
    let image = new Image();
    // 解决跨域问题
    image.crossOrigin = "anonymous";
    //图片地址
    image.src = imgUrl;
    // image.onload为异步加载
    image.onload = () => {
      let canvas = document.createElement("canvas");
      canvas.width = image.width;
      canvas.height = image.height;
      let context = canvas.getContext("2d");
      context.drawImage(image, 0, 0, image.width, image.height);
      //转成base64
      let dataurl = canvas.toDataURL("image/png", 0.8);
      //返回
      resolve(dataurl);
    };
  });
}

/**
 * 导出word,支持图片
 * @param {Object} tempDocxPath 模板文件路径
 * @param {Object} wordData 导出数据
 * @param {Object} fileName 导出文件名
 * @param {Object} imgSize 预留，自定义图片尺寸 => 暂没使用
 */
export const exportWordDocx = (tempDocxPath, wordData, fileName, imgSize) => {
  // 这里要引入处理图片的插件
  var ImageModule = docxtemplaterImage;

  JSZipUtils.getBinaryContent(tempDocxPath, function (error, content) {
    console.log(wordData);
    if (error) {
      throw error;
    }

    let opts = {};
    opts = {
      centered: true, //图像是否居中，true：在word中图片居中
      getImage: (item) => {
        // 将base64转成ArrayBuffer
        return base64DataURLToArrayBuffer(item);
      },
      //自定义指定图像大小，此处可动态调试各别图片的大小
      getSize: (img, tagValue, tagName) => {
        return [200, 150];
      },
    };

    // 创建一个PizZip实例，内容为模板的内容
    let zip = new PizZip(content);
    // 创建并加载docxtemplater实例对象
    let doc = new Docxtemplater(zip, {
      modules: [new ImageModule(opts)],
    });
    // doc.attachModule(new ImageModule(opts));
    // 图片处理
    // 如果使用对象去解析 wordData: { obj: {...} }；可能需要配置parser => new Docxtemplater(zip, { parser: expressionParser }); 具体见官网
    // doc.loadZip(zip);
    // 设置模板变量的值
    doc.setData(wordData);
    try {
      // 用模板变量的值替换所有模板变量
      doc.render();
    } catch (error) {
      // 抛出异常
      let e = {
        message: error.message,
        name: error.name,
        stack: error.stack,
        properties: error.properties,
      };
      console.log(JSON.stringify({ error: e }));
      throw error;
    }
    // 生成一个代表docxtemplater对象的zip文件（不是一个真实的文件，而是在内存中的表示）
    let out = doc.getZip().generate({
      type: "blob",
      mimeType:
        "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
    });
    // 将目标文件对象保存为目标类型的文件，并命名
    saveAs(out, fileName);
  });
};
